model = My_VGG16(class_num=2)
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model.to(device)
model.train() #修改6：用BN，training=True切记
